python printing identity matrix